#ifndef UTIL_HPP
#define UTIL_HPP

#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>

class Util
{
public:
    static int ReadLine(int sock, std::string& out)
    {
        char ch = 'X';
        while(ch != '\n')
        {
            ssize_t s = recv(sock, &ch, 1, 0);
            if(s > 0)
            {
                if(ch == '\r')
                {
                    // 1. \r->\n  2. \r\n->\n
                    // 我们需要先查看一下\r的后面字符，但并不取走，因为有可能是下一行的开头
                    recv(sock, &ch, 1, MSG_PEEK);   // 窥探下一个字符
                    if(ch == '\n')
                    {
                        recv(sock, &ch, 1, 0);
                    }
                    else
                    {
                        ch = '\n';
                    }
                }
                // 1.普通字符 2.\n
                out.push_back(ch);
            }
            else if(s == 0)
            {
                return 0;
            }
            else
            {
                return -1;
            }
        }
        return out.size();
    }

    static bool CutString(const std::string& target, std::string& key_out, std::string& value_out, std::string sep)
    {
        size_t pos = target.find(sep);
        if(pos != std::string::npos)
        {
            key_out = target.substr(0, pos);
            value_out = target.substr(pos+sep.size());
            return true;
        }
        return false;
    }
};

#endif